import torch
print(torch.cuda.is_available())    # 查看CUDA是否可用
print(torch.cuda.device_count())    # 查看可用的CUDA数量
print(torch.version.cuda)           # 查看CUDA的版本号
levels = torch.tensor(
    [[0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4]])



